Using Python for loop to iterate over a list with index